package com.datastructure.test.threeoftotal.self;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

public class ThreeOfArray {
    public List<List<Integer>> getThreeOfArray(int[] data){
        List<List<Integer>> res = new ArrayList<>();
        Arrays.sort(data);
        for (int i = 0; i < data.length; i++) {
            if(data[i]>0){
                break;
            }
            if(i>0&&(data[i-1]==data[i])){
                continue;
            }
            int start = i+1;
            int end = data.length-1;
            while (start<end){
                int sum = data[i]+data[start]+data[end];
                if(sum==0){
                    List<Integer> item = new ArrayList<>();
                    item.add(data[i]);
                    item.add(data[start]);
                    item.add(data[end]);
                    res.add(item);
                    while (start<end&&(data[start+1]==data[start])){
                        start++;
                    }
                    while (start<end&&(data[end-1]==data[end])){
                        end--;
                    }
                    start++;
                    end--;
                }else if(sum>0) {
                    end--;
                } else {
                    start++;
                }
            }
        }
        return res;
    }
}
